package com.xuecheng.content.service.impl;

import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.core.conditions.update.LambdaUpdateWrapper;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.xuecheng.api.content.model.dto.CourseBaseDTO;
import com.xuecheng.api.content.model.dto.TeachplanDTO;
import com.xuecheng.api.content.model.vo.TeachplanVO;
import com.xuecheng.common.domain.code.CommonErrorCode;
import com.xuecheng.common.enums.common.CommonEnum;
import com.xuecheng.common.enums.content.CourseAuditEnum;
import com.xuecheng.common.enums.content.TeachPlanEnum;
import com.xuecheng.common.exception.ExceptionCast;
import com.xuecheng.common.util.StringUtil;
import com.xuecheng.content.common.constant.ContentErrorCode;
import com.xuecheng.content.convert.TeachplanConvert;
import com.xuecheng.content.entity.CourseBase;
import com.xuecheng.content.entity.Teachplan;
import com.xuecheng.content.entity.ext.TeachplanNode;
import com.xuecheng.content.mapper.TeachplanMapper;
import com.xuecheng.content.service.CourseBaseService;
import com.xuecheng.content.service.TeachplanService;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional;
import org.springframework.util.CollectionUtils;
import org.springframework.util.ObjectUtils;
import org.springframework.web.bind.annotation.RequestBody;

import java.util.ArrayList;
import java.util.List;

/**
 * <p>
 * 课程计划 服务实现类
 * </p>
 *
 * @author itcast
 */
@Slf4j
@Service
public class TeachplanServiceImpl extends ServiceImpl<TeachplanMapper, Teachplan> implements TeachplanService {
    @Autowired
    private CourseBaseService courseBaseService;

    /*
     * 业务分析：
     *   1.判断关键数据
     *       courseId companyId
     *   2.判断业务数据
     *       课程基础信息
     *           判断是否存在
     *           判断是否是同一家教学机构
     *           判断是否删除
     *   3.根据Courseid查询课程计划数据
     *   4.通过java的递归生成课程计划树形结构
     *   5.将数据转为dto并返回
     * */
    public TeachplanDTO getTreeNodes(Long courseId, Long companyId) {

        // 1.判断关键数据
        //      courseId companyId
        if (ObjectUtils.isEmpty(courseId)||
                ObjectUtils.isEmpty(companyId)
        ) {
            ExceptionCast.cast(CommonErrorCode.E_100101);
        }

        //  2.判断业务数据
        //      课程基础信息
        //          判断是否存在
        //          判断是否是同一家教学机构
        //          判断是否删除
        LambdaQueryWrapper<CourseBase> baseQueryWrapper = new LambdaQueryWrapper<>();
        baseQueryWrapper.eq(CourseBase::getId, courseId);
        baseQueryWrapper.eq(CourseBase::getCompanyId, companyId);

        CourseBase courseBase = courseBaseService.getOne(baseQueryWrapper);

        if (ObjectUtils.isEmpty(courseBase)) {
            ExceptionCast.cast(ContentErrorCode.E_120013);
        }

        Integer status = courseBase.getStatus();
        if (!(CommonEnum.USING_FLAG.getCodeInt().equals(status))) {
            ExceptionCast.cast(ContentErrorCode.E_120021);
        }

        //  3.根据Courseid查询课程计划数据
        TeachplanMapper baseMapper = this.getBaseMapper();
        List<TeachplanNode> nodes = baseMapper.selectTreeNodesByCourseId(courseId);


        //  4.通过java的递归生成课程计划树形结构

        TeachplanDTO teachplanDTO = null;
        if (CollectionUtils.isEmpty(nodes)) {
            teachplanDTO = new TeachplanDTO();
        } else {

            TeachplanNode rootNode = nodes.remove(0);

            generateTreeNodes(rootNode,nodes);

            teachplanDTO = TeachplanConvert.INSTANCE.node2dto(rootNode);
        }


        //  5.将数据转为dto并返回
        return teachplanDTO;
    }



    private void generateTreeNodes(TeachplanNode parentNode, List<TeachplanNode> nodes) {

        // 1.判断父级集合是否为空，如果为空需要创建
        if (CollectionUtils.isEmpty(parentNode.getChildrenNodes())) {
            parentNode.setChildrenNodes(new ArrayList<>(8));
        }

        // 2.循环遍历集合生成树形结构
        for (TeachplanNode node : nodes) {

            // 2.1 判断node是否和parentnode有关联，如果有关联将node存放到parentNode集合中
            if (ObjectUtils.nullSafeEquals(parentNode.getId(),node.getParentid())) {

                parentNode.getChildrenNodes().add(node);

                // 2.2 生成node集合数据
                // 如果是第三级课程计划，无需递归
                if (!(TeachPlanEnum.THIRD_LEVEL.equals(node.getGrade()))) {
                    generateTreeNodes(node,nodes);
                }
            }
        }

    }


    /*
     * 业务分析：
     *   主体方法
     *       1.判断业务操作
     *           判断是创建还是修改：teachplanId
     *           如果有:修改
     *           如果没有:创建
     *
     *       2.执行创建和修改操作并返回dto数据
     *
     *       3.返回结果数据dto
     * */
    @Transactional
    public TeachplanDTO createOrModifyTeachplan(TeachplanDTO dto, Long companyId) {

        TeachplanDTO resutlDTO = null;
        //1.判断业务操作
        //     判断是创建还是修改：teachplanId
        //     如果有:修改
        //     如果没有:创建
        // 2.执行创建和修改操作并返回dto数据
        if (ObjectUtils.isEmpty(dto.getTeachPlanId())) {
            resutlDTO = createTeachplan(dto, companyId);
        } else {
            resutlDTO = modifyTeachplan(dto, companyId);
        }

        // 3.返回结果数据dto
        return resutlDTO;
    }


    /*
    * 业务分析：
    *    1.判断关键数据
         2.判断业务数据
             课程基础信息
                 判断是否存在
                 判断是否同一家机构
                 判断是否删除
                 判断课程基础信息审核状态：未提交和审核未通过
             课程计划
                 判断是否存在
         3.修改课程计划数据
              pname starttime endtime is_preview
         4.判断修改后的结果
         5.将数据库最新的数据转为dto并返回
    * */
    private TeachplanDTO modifyTeachplan(TeachplanDTO dto, Long companyId) {

        // 1.判断关键数据
        // 2.判断业务数据
        //      课程基础信息
        //          判断是否存在
        //          判断是否同一家机构
        //          判断是否删除
        //          判断课程基础信息审核状态：未提交和审核未通过
        CourseBaseDTO baseDTO = verifyTeachplanMsg(dto, companyId);


        //      课程计划
        //          判断是否存在
        LambdaQueryWrapper<Teachplan> queryWrapper = new LambdaQueryWrapper<>();
        queryWrapper.eq(Teachplan::getId, dto.getTeachPlanId());
        int count = this.count(queryWrapper);
        if (count < 1) {
            ExceptionCast.cast(ContentErrorCode.E_120402);
        }


        //  3.修改课程计划数据
        //       pname starttime endtime is_preview
        LambdaUpdateWrapper<Teachplan> updateWrapper = new LambdaUpdateWrapper<>();
        updateWrapper.set(Teachplan::getPname, dto.getPname());
        updateWrapper.set(Teachplan::getStartTime, dto.getStartTime());
        updateWrapper.set(Teachplan::getEndTime, dto.getEndTime());
        updateWrapper.set(Teachplan::getIsPreview, dto.getIsPreview());

        updateWrapper.eq(Teachplan::getId, dto.getTeachPlanId());


        //  4.判断修改后的结果
        boolean result = this.update(updateWrapper);

        if (!result) {
            ExceptionCast.cast(ContentErrorCode.E_120407);
        }

        //  5.将数据库最新的数据转为dto并返回
        Teachplan po = this.getById(dto.getTeachPlanId());

        TeachplanDTO resultDTO = TeachplanConvert.INSTANCE.entity2dto(po);

        return resultDTO;
    }

    /*
     * 主体方法业务分析：
     *   1.判断关键数据
     *   2.判断业务数据
     *       课程基础信息
     *           判断是否存在
     *           判断是否同一家机构
     *           判断是否删除
     *           判断课程基础信息审核状态：未提交和审核未通过
     *   3.获得父级数据并进行关联
     *       parentid
     *       grade：子级grade是父级grade+1
     *       orderby：获得父级课程计划的子级数据个数+1（bug）
     *   4.将dto转为po并保存
     *   5.判断保存后的结果
     *   6.将数据库最新数据转为dto并返回
     *
     * */
    private TeachplanDTO createTeachplan(TeachplanDTO dto, Long companyId) {

        CourseBaseDTO courseBase = verifyTeachplanMsg(dto, companyId);


        // 3.获得父级数据并进行关联
        //     parentid
        //     grade：子级grade是父级grade+1
        //     orderby：获得父级课程计划的子级数据个数+1（bug）
        Teachplan teachplan = TeachplanConvert.INSTANCE.dto2entity(dto);

        Teachplan parentNode =  generateParentNode(dto,courseBase);

        teachplan.setParentid(parentNode.getId());
        teachplan.setGrade(parentNode.getGrade()+1);

        // select count(1) from teachplan where parentId = parentNode.getId();
        LambdaQueryWrapper<Teachplan> queryWrapper = new LambdaQueryWrapper<>();
        queryWrapper.eq(Teachplan::getParentid, parentNode.getId());

        int count = this.count(queryWrapper);
        teachplan.setOrderby(count+1);

        // 4.将dto转为po并保存
        boolean result = this.save(teachplan);

        // 5.判断保存后的结果
        if (!result) {
            ExceptionCast.cast(ContentErrorCode.E_120407);
        }


        // 6.将数据库最新数据转为dto并返回
        Teachplan po = this.getById(teachplan.getId());

        TeachplanDTO resultDTO = TeachplanConvert.INSTANCE.entity2dto(po);

        return resultDTO;
    }

    private CourseBaseDTO verifyTeachplanMsg(TeachplanDTO dto, Long companyId) {
        //1.判断关键数据
        if (ObjectUtils.isEmpty(companyId)||
                StringUtil.isBlank(dto.getPname())||
                ObjectUtils.isEmpty(dto.getCourseId())
        ) {
            ExceptionCast.cast(CommonErrorCode.E_100101);
        }


        // 2.判断业务数据
        //     课程基础信息
        //         判断是否存在
        //         判断是否同一家机构
        //         判断是否删除
        //         判断课程基础信息审核状态：未提交和审核未通过
        CourseBaseDTO courseBase = courseBaseService.getCourseBaseById(dto.getCourseId(), companyId);

        String auditStatus = courseBase.getAuditStatus();
        if (CourseAuditEnum.AUDIT_PASTED_STATUS.getCode().equals(auditStatus)||
                CourseAuditEnum.AUDIT_COMMIT_STATUS.getCode().equals(auditStatus)||
                CourseAuditEnum.AUDIT_PUBLISHED_STATUS.getCode().equals(auditStatus)
        ) {
            ExceptionCast.cast(ContentErrorCode.E_120015);
        }
        return courseBase;
    }


    /*
     * 业务分析：
     *   1.判断新增数据是二级还是三级课程计划
     *       判断新增数据是否有parentId值
     *           如果有：三级课程计划
     *           如果没有：二级课程计划
     *   2.获得二级课程计划的父级数据
     *       查询数据库获得一级课程计划：courseId、parentid=0
     *           如果没有：自动创建并保存数据--数据来源 coursebase
     *           如果有：直接返回
     *   3.获得三级课程计划的父级数据
     *       查询数据库获得二级课程计划：parentId
     *           如果没有：直接抛出异常
     *           如果有：直接返回
     * */
    private Teachplan generateParentNode(TeachplanDTO dto, CourseBaseDTO courseBase) {

        //1.判断新增数据是二级还是三级课程计划
        //     判断新增数据是否有parentId值
        //         如果有：三级课程计划
        //         如果没有：二级课程计划
        Long parentid = dto.getParentid();
        if (ObjectUtils.isEmpty(parentid)) {
            // 2.获得二级课程计划的父级数据
            //     查询数据库获得一级课程计划：courseId、parentid=0
            //         如果没有：自动创建并保存数据--数据来源 coursebase
            //         如果有：直接返回
            LambdaQueryWrapper<Teachplan> queryWrapper = new LambdaQueryWrapper<>();
            queryWrapper.eq(Teachplan::getCourseId, courseBase.getCourseBaseId());
            queryWrapper.eq(Teachplan::getParentid, TeachPlanEnum.FIRST_PARENTID_FLAG);

            Teachplan rootParent = this.getOne(queryWrapper);

            if (ObjectUtils.isEmpty(rootParent)) {

                rootParent = new Teachplan();
                rootParent.setPname(courseBase.getName());
                rootParent.setParentid(new Long(TeachPlanEnum.FIRST_PARENTID_FLAG));
                rootParent.setGrade(TeachPlanEnum.FIRST_LEVEL);
                rootParent.setDescription(courseBase.getDescription());
                rootParent.setOrderby(TeachPlanEnum.FIRST_LEVEL);
                rootParent.setCourseId(courseBase.getCourseBaseId());
                boolean save = this.save(rootParent);

                if (!save) {
                    ExceptionCast.cast(ContentErrorCode.E_120415);
                }
            }

            return rootParent;

        } else {
            // 3.获得三级课程计划的父级数据
            //     查询数据库获得二级课程计划：parentId
            //         如果没有：直接抛出异常
            //         如果有：直接返回
            LambdaQueryWrapper<Teachplan> queryWrapper = new LambdaQueryWrapper<>();
            queryWrapper.eq(Teachplan::getId, parentid);

            Teachplan teachplan = this.getOne(queryWrapper);

            if (ObjectUtils.isEmpty(teachplan)) {
                ExceptionCast.cast(ContentErrorCode.E_120408);
            }

            return teachplan;
        }
    }

}
